//
//  badge.m
//  CreditAddressBook
//
//  Created by mosu on 2016/9/26.
//  Copyright © 2016年 Lee. All rights reserved.
//

#import "badge.h"
#define TabbarItemNums 5.0    //tabbar的数量

@implementation badge

- (void)showBadgeOnItemIndex:(int)index{
     
     //移除之前的小红点
     [self removeBadgeOnItemIndex:index];
     
     //新建小红点
     UIView *badgeView = [[UIView alloc]init];
     badgeView.tag = 888 + index;
     badgeView.layer.cornerRadius = 5;
     badgeView.backgroundColor = [UIColor redColor];
     CGRect tabFrame = self.frame;
     
     //确定小红点的位置
     float percentX = (index +0.6) / TabbarItemNums;
     CGFloat x = ceilf(percentX * tabFrame.size.width);
     CGFloat y = ceilf(0.1 * tabFrame.size.height);
     badgeView.frame = CGRectMake(x, y, 10, 10);
     [self addSubview:badgeView];
     
}

- (void)hideBadgeOnItemIndex:(int)index{
     
     //移除小红点
     [self removeBadgeOnItemIndex:index];
     
}

- (void)removeBadgeOnItemIndex:(int)index{
     
     //按照tag值进行移除
     for (UIView *subView in self.subviews) {
          
          if (subView.tag == 888+index) {
               
               [subView removeFromSuperview];
               
          }
     }
}



/*
// Only override drawRect: if you perform custom drawing.
// An empty implementation adversely affects performance during animation.
- (void)drawRect:(CGRect)rect {
    // Drawing code
}
*/

@end
